"""
@-*- coding: utf-8 -*-
@ python：python 3.8
@ 创建人员：ylx
@ 创建时间：2025/2/28 下午5:01
@ 内容：
"""
import os
import yaml


#读取
def read_yaml(key):
    with open(os.getcwd() + '/extract.yaml') as f:
        #先打开这个yaml文件，使用命令获取文件当前所在目录，然后+文件名称来拼接获取这个文件的路径即可
        #默认的mode 是只读模式
        value = yaml.load(stream=f, Loader=yaml.FullLoader)  #读取yaml文件的值，并返回
        return value[key]


#写入
def write_yaml(data):
    with open(os.getcwd() + '/extract.yaml', mode="a", encoding="utf-8") as f:
        #先打开这个yaml文件，使用命令获取项目的根目录，然后+文件名称来拼接获取这个文件的路径即可
        #打开后默认的mode是只读，写入有2种方式 w覆盖写  a追加写
        yaml.dump(data, stream=f, allow_unicode=True)


#清空
def clean_yaml():
    with open(os.getcwd() + '/extract.yaml', mode="w", encoding="utf-8") as f:
        #先打开这个yaml文件，使用命令获取项目的根目录，然后+文件名称来拼接获取这个文件的路径即可
        #打开后默认的mode是只读，写入有2种方式 w覆盖写  a追加写
        f.truncate()


def read_testcase(yaml_name):
    with open(os.getcwd() + '/testcase/' + yaml_name) as f:
        #先打开这个yaml文件，使用命令获取项目的根目录，然后+文件名称来拼接获取这个文件的路径即可
        #默认的mode 是只读模式
        value = yaml.load(stream=f, Loader=yaml.FullLoader)  #读取yaml文件的值，并返回
        return value


if __name__ == "__main__":
    print(os.getcwd())